package com.wangxiang.common.utils;

import com.github.pagehelper.PageInfo;
import org.springframework.stereotype.Component;

import java.util.HashMap;


@Component
public class PageGenerate {
    public PageEntity generate(PageInfo<?> pageInfo){
        PageEntity pageEntity = new PageEntity();
//        System.out.println("工具接受"+pageInfo);
        pageEntity.setTotalPage(pageInfo.getPages());
        pageEntity.setTotalCount(pageInfo.getTotal());
        pageEntity.setPageNum(pageInfo.getPageNum());
        pageEntity.setPageSize(pageInfo.getPageSize());
        pageEntity.setList(pageInfo.getList());
        return pageEntity;
    }
    public PageEntity generate(PageInfo<?> pageInfo, HashMap<String,Object> map){
        PageEntity pageEntity = new PageEntity();
//        System.out.println("工具接受"+pageInfo);
        pageEntity.setTotalPage(pageInfo.getPages());
        pageEntity.setTotalCount(pageInfo.getTotal());
        pageEntity.setPageNum(pageInfo.getPageNum());
        pageEntity.setPageSize(pageInfo.getPageSize());
        pageEntity.setList(pageInfo.getList());
        pageEntity.setMap(map);
        return pageEntity;
    }

}
